/*
 * Copyright (c) 2024 Huawei Device Co., Ltd.
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

import { ir, native, HomeCheckIrHint, getIrLocationOfToken } from 'qvog-dsl';

const SEVERITY = 2;
const DOC_PATH = 'docs/array-definition-check-dsl.md';
const DESCRIPTION = 'Array type definition is not correct.';

export default native<HomeCheckIrHint>({ use: 'ir' })()
    .match((stmt): stmt is ir.ArkAssignStmt => {
        return stmt instanceof ir.ArkAssignStmt;
    })
    .when((stmt) => {
        let rightOp = stmt.getRightOp();
        if (!rightOp || !(rightOp instanceof ir.ArkNewArrayExpr)) {
            return false;
        }
        let text = stmt.getOriginalText() ?? '';
        return isGenericArray(text);
    })
    .report((stmt, { api }): void => {
        api.report({
            location: getIrLocationOfToken(stmt, "Array"),
            severity: SEVERITY,
            description: DESCRIPTION,
            docPath: DOC_PATH,
        })
    });

function isGenericArray(text: string): boolean {
    let matchArr = text.match(/:([\s\S]*?)=/);
    if (matchArr !== null) {
        let matchTags = matchArr[1].match(/Array|<|>/g);
        return matchTags !== null && matchTags.includes('Array') && matchTags.includes('<') && matchTags.includes('>');
    }
    return false;
}